# 题目描述：
# 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
#
# 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
#
# 你可以按任意顺序返回答案。
#
#  
#
# 示例 1：
#
# 输入：nums = [2,7,11,15], target = 9
# 输出：[0,1]
# 解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
# 示例 2：
#
# 输入：nums = [3,2,4], target = 6
# 输出：[1,2]
# 示例 3：
#
# 输入：nums = [3,3], target = 6
# 输出：[0,1]
#
# 来源：力扣（LeetCode）
# 链接：https://leetcode-cn.com/problems/two-sum
# 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。


class Solution:
    # 暴力法
    # 时间复杂度O(n^2) 空间复杂度O(1)
    # 两层循环，故时间复杂度O(n^2)；过程中用了i、j，空间复杂度O(1)。
    def two_sum(self, nums, target):
        for i in range(len(nums) - 1):
            for j in range(i + 1, len(nums)):
                if nums[i] + nums[j] == target:
                    return [i, j]

    # 能否想出时间复杂度小于n^2的解法？
    # 用哈希表存储一次遍历的一些中间值
    def two_sum_solution2(self, nums, target):
        temp = {nums[0]: 0}
        for i in range(1, len(nums)):
            m = target - nums[i]
            if m in temp.keys():
                return [temp[m], i]
            temp[nums[i]] = i


    def main(self):
        nums = [3, 2, 4]
        target = 6
        # res = self.twoSum(nums, target)
        res = self.two_sum_solution2(nums, target)
        print(res)


if __name__ == '__main__':
    Solution().main()
